package com.dmlll.servlet;

import com.jspsmart.upload.File;
import com.jspsmart.upload.SmartUpload;
import com.jspsmart.upload.SmartUploadException;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.jsp.JspFactory;
import javax.servlet.jsp.PageContext;
import java.io.IOException;
import java.net.URLEncoder;

@WebServlet(urlPatterns = "/upload")
public class UploadServlet extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        String filename = req.getParameter("filename");
        resp.setContentType("application/octet-stream");
        resp.addHeader("Content-Disposition","attachment;filename="+filename);
        req.getRequestDispatcher("/upload/"+filename).forward(req,resp);
        resp.flushBuffer();
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        try {
            //创建上传文件的对象
            SmartUpload smartUpload = new SmartUpload();
            //初始化
            PageContext pageContext = JspFactory.getDefaultFactory().getPageContext(this,req,resp,null,false,1024,true);
            smartUpload.initialize(pageContext);
            //设置编码
            smartUpload.setCharset("utf-8");
            //上传
            smartUpload.upload();

            //获取文件信息
            File file = smartUpload.getFiles().getFile(0);
            String fileName = file.getFileName();
            String contentType = file.getContentType();
            //指定上传路径
            String s = "/upload/" + fileName;
            //保存文件
            file.saveAs(s,File.SAVEAS_VIRTUAL);
            //跳转页面
            req.setAttribute("filename",fileName);
            req.getRequestDispatcher("success.jsp").forward(req,resp);
        } catch (SmartUploadException e) {
            e.printStackTrace();
        }
    }
}
